add_executable(allreduce allreduce.cpp)
target_include_directories(allreduce PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/..)
target_link_libraries(allreduce pthread torch_cpu)
if(USE_CUDA)
  target_link_libraries(allreduce torch_cuda)
endif()
